check armstrong number in python